import warnings
warnings.filterwarnings("ignore")
from ultralytics import YOLO

if __name__ == "__main__":
    model = YOLO("yolo11n.pt")  # Load a pretrained model
    model.train(data=r"E:\Task\yolov11\datasets\mask\data.yaml", 
                cfg=r"E:\Task\yolov11\datasets\mask\default.yaml")
